diff --git a/tests/codeigniter/helpers/language_helper_test.php b/tests/codeigniter/helpers/language_helper_test.php
index 1fe0ef1..176da68 100644
--- a/tests/codeigniter/helpers/language_helper_test.php
+++ b/tests/codeigniter/helpers/language_helper_test.php
@@ -10,7 +10,7 @@
 		$this->ci_instance_var('lang', $lang);
 
 		$this->assertFalse(lang(1));
-		$this->assertEquals('<label for="foo"></label>', lang(1, 'foo'));
+		$this->assertEquals('<label for="foo" class="bar"></label>', lang(1, 'foo', array('class' => 'bar')));
 	}
 
 }
\ No newline at end of file